package top.whysu.gps.utils;

import com.github.pagehelper.Page;
import com.github.pagehelper.PageHelper;
import org.apache.commons.lang3.StringUtils;
import top.whysu.gps.conf.SortFieldConfig;
import top.whysu.gps.dto.page.BasePagerDTO;
import top.whysu.gps.dto.page.OrderByDTO;
import top.whysu.gps.dto.page.PagenationDataDTO;

import java.util.List;
import java.util.Map;

/**
 * pageHelper分页工具类
 */
public class PageUtils {
    private static SortFieldConfig sortFieldConfig = SpringUtil.getBean(SortFieldConfig.class);
    /**
     * 升序
     */
    private static final String ASC = "asc";
    /**
     * 降序
     */
    private static final String DESC = "desc";
    /**
     * 逗号
     */
    private static final String COMMA = ",";
    /**
     * 空格
     */
    private static final String BLANK = " ";

    /**
     * 获得起始值
     */
    public static int getRowIndex(int pageNum, int pageSize) {
        return (pageNum > 0) ? (pageNum -1) * pageSize : 0;
    }

    /**
     * 将查询结果转成分页对象
     */
    public static <T> PagenationDataDTO<T> toPagedResult(List<T> datas) {
        if (datas != null) {
            if (datas instanceof Page) {
                Page page = (Page) datas;
                return new PagenationDataDTO<T>(page.getTotal(), page.getPages(), page.getPageNum(), page.getPageSize(), page.getResult());
            } else {
                return new PagenationDataDTO<T>(1, datas.size(), 1, datas.size(), datas);
            }
        } else {
            return new PagenationDataDTO<T>(1, 0, 1, 0, null);
        }
    }

    public static <PO, DTO> PagenationDataDTO<DTO> toPagedResult(List<PO> datas, Class<DTO> clazz) {
        if (datas instanceof Page) {
            Page page = (Page) datas;
            return new PagenationDataDTO<DTO>(page.getTotal(), page.getPages(), page.getPageNum(), page.getPageSize(),
                    ConverterUtils.po2dto(page.getResult(), clazz));
        } else {
            return new PagenationDataDTO<DTO>(1, datas.size(), 1, datas.size(),
                    ConverterUtils.po2dto(datas, clazz));
        }
    }

    /**
     * 查询结果转换为分页对象-传入分页信息
     *
     * @param datas
     * @param page
     * @param clazz
     * @param <PO>
     * @param <DTO>
     * @return
     */
    public static <PO, DTO> PagenationDataDTO<DTO> toPagedResult(List<DTO> datas, Page page, Class<DTO> clazz) {
        if (page instanceof Page) {
            return new PagenationDataDTO<DTO>(page.getTotal(), page.getPages(), page.getPageNum(), page.getPageSize(), datas);
        } else {
            return new PagenationDataDTO<DTO>(1, datas.size(), 1, datas.size(), ConverterUtils.po2dto(datas, clazz));
        }
    }

    /**
     * 分页查询
     */
    public static <T> Page<T> paging(BasePagerDTO basePager, String sortKey) {
        Page<T> page;
        OrderByDTO[] orderBys = basePager.getOrders();
        StringBuilder sb = new StringBuilder();
        if (StringUtils.isNotBlank(sortKey)) {
            if (null != orderBys && orderBys.length > 0) {
                Map<String, String> map = sortFieldConfig.getSortFieldMap().get(sortKey);
                if (null == map || map.isEmpty()) {
//                    throw new BusinessException(MessageCommonConstant.QUERY_FAIL);
                }
                for (OrderByDTO orderBy : orderBys) {
                    if (null == orderBy) {
//                        throw new BusinessException(MessageCommonConstant.QUERY_FAIL);
                    }
                    // 从配置文件中获取
                    String sort = map.get(orderBy.getSort());
                    if (StringUtils.isNotBlank(sort)) {
                        String order = DESC;
                        if (StringUtils.isNotBlank(orderBy.getOrder())) {
                            order = ASC.equalsIgnoreCase(orderBy.getOrder()) ? ASC : DESC;
                        }
                        sb.append(sort).append(BLANK).append(order).append(COMMA);
                    } else {
//                        throw new BusinessException(MessageCommonConstant.QUERY_FAIL);
                    }
                }
            }
        }

        if (sb.length() > 0) {
            page = PageHelper.startPage(basePager.getPageNum(), basePager.getPageSize(), sb.deleteCharAt(sb.length() - 1).toString());
        } else {
            page = PageHelper.startPage(basePager.getPageNum(), basePager.getPageSize());
        }
        return page;
    }

    public static <T> Page<T> paging(BasePagerDTO basePager) {
        return paging(basePager, null);
    }
}
